# @Author : wangyuchen
# @Time : 2021-05-13 10:45


from datetime import datetime

# mean and std of training dataset[0.485, 0.456, 0.406], [0.229, 0.224, 0.225]
TRAIN_MEAN = (0.485, 0.456, 0.406)
TRAIN_STD = (0.229, 0.224, 0.225)
VAL_MEAN = (0.485, 0.456, 0.406)
VAL_STD = (0.229, 0.224, 0.225)
TEST_MEAN = (0.485, 0.456, 0.406)
TEST_STD = (0.229, 0.224, 0.225)

# directory to save weights file
CHECKPOINT_PATH = 'checkpoint'

# total training epoches
EPOCH = 50
MILESTONES = [5, 12, 18, 23]

# time of we run the script
TIME_NOW = datetime.now().strftime('%A_%d_%B_%Y_%Hh_%Mm_%Ss')

# tensorboard log dir
LOG_DIR = 'runs'

# save weights file per SAVE_EPOCH epoch
SAVE_EPOCH = 10

ANNO_PATH = 'val_anno.txt'
PRED_PATH = '181250147-val-{epoch}.txt'
TEST_RES_PATH = '181250147-{net}.txt'
